home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Ubuntu 9.10 PL / karmelkowy-koliberek-desktop-9.10-i386-PL.iso / casper / filesystem.squashfs / usr / share / gufw / model / FrontEnd.py < prev    next >
Encoding:
Python Source  |  2009-10-25  |  10.5 KB  |  304 lines
  1. # Gufw 9.10.4 - http://gufw.tuxfamily.org
  2. # Copyright (C) 2009 Marcos Alvarez Costales
  3. #
  4. # Gufw is free software; you can redistribute it and/or modify
  5. # it under the terms of the GNU General Public License as published by
  6. # the Free Software Foundation; either version 3 of the License, or
  7. # (at your option) any later version.
  8. # 
  9. # Gufw is distributed in the hope that it will be useful,
  10. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12. # GNU General Public License for more details.
  13. # 
  14. # You should have received a copy of the GNU General Public License
  15. # along with Gufw; if not, see http://www.gnu.org/licenses for more
  16. # information.
  17.  
  18.  
  19. import commands
  20. from Variable import Variable
  21.  
  22.  
  23. class Frontend:
  24.  
  25.     def __init__(self):
  26.         self.variable = Variable()
  27.  
  28.        
  29.     # Return status ufw
  30.     def set_status(self, status):
  31.         commands.getstatusoutput(self.variable.get_command(status))
  32.         return status
  33.  
  34.  
  35.     # Get status ufw
  36.     def get_status(self):
  37.         status = commands.getstatusoutput(self.variable.get_command("status"))
  38.         if status[1].find("Status: active") != -1:
  39.             return self.variable.get_constant("enabled")
  40.         else:
  41.             return self.variable.get_constant("disabled")
  42.  
  43.  
  44.     # Return status ufw
  45.     def set_default(self, default):
  46.         commands.getstatusoutput(self.variable.get_command(default))
  47.         return default
  48.  
  49.  
  50.     # Get status ufw
  51.     def get_default(self):
  52.         default = commands.getstatusoutput(self.variable.get_command("search_policy"))
  53.         if default[1].find("ACCEPT") != -1:
  54.             return self.variable.get_constant("allow")
  55.         elif default[1].find("DROP") != -1:
  56.             return self.variable.get_constant("deny")
  57.         elif default[1].find("REJECT") != -1:
  58.             return self.variable.get_constant("reject")
  59.  
  60.  
  61.     # Get actual rules (NOT SPLIT BY ACTION)
  62.     def get_rules(self):
  63.         rules        = commands.getstatusoutput(self.variable.get_command("status"))
  64.         rule_lines   = rules[1].split("\n")
  65.         return_rules = []
  66.         
  67.         for descomponent_rules in rule_lines:
  68.             
  69.             if descomponent_rules.find(self.variable.get_constant("allow_upper"))  != -1 or \
  70.                descomponent_rules.find(self.variable.get_constant("deny_upper"))   != -1 or \
  71.                descomponent_rules.find(self.variable.get_constant("limit_upper"))  != -1 or \
  72.                descomponent_rules.find(self.variable.get_constant("reject_upper")) != -1:
  73.                 
  74.                 return_rules.append(descomponent_rules)
  75.                 
  76.         return return_rules
  77.  
  78.  
  79.     # Return status ufw
  80.     def set_ufw_log(self, status_ufw_log):
  81.         commands.getstatusoutput(self.variable.get_command(status_ufw_log))
  82.         return status_ufw_log
  83.  
  84.  
  85.     # Get status ufw
  86.     def get_ufw_log(self):
  87.         status = commands.getstatusoutput(self.variable.get_command("status_verbose"))
  88.         if status[1].find("Logging: on (high)") != -1:
  89.             return self.variable.get_constant("ufw_log_high")
  90.         elif status[1].find("Logging: on (medium)") != -1:
  91.             return self.variable.get_constant("ufw_log_medium")
  92.         elif status[1].find("Logging: on (low)") != -1:
  93.             return self.variable.get_constant("ufw_log_low")
  94.         elif status[1].find("Logging: on (full)") != -1:
  95.             return self.variable.get_constant("ufw_log_full")
  96.         else:
  97.             return self.variable.get_constant("ufw_log_off")
  98.  
  99.  
  100.     # Return status ufw
  101.     def set_gufw_log(self, status_gufw_log):
  102.         if not self.variable.dev:
  103.             status_gufw_log_aux = status_gufw_log
  104.         else:
  105.             status_gufw_log_aux = status_gufw_log + "_dev"
  106.             
  107.         commands.getstatusoutput(self.variable.get_command(status_gufw_log_aux))
  108.         return status_gufw_log
  109.  
  110.  
  111.     # Get status ufw
  112.     def get_gufw_log(self):
  113.         if not self.variable.dev:
  114.             command = commands.getstatusoutput(self.variable.get_command("cfg_gufw_log"))
  115.         else:
  116.             command = commands.getstatusoutput(self.variable.get_command("cfg_gufw_log_dev"))
  117.             
  118.         if command[0] == 0:
  119.             return self.variable.get_constant("gufw_log_on")
  120.         else:
  121.             return self.variable.get_constant("gufw_log_off")
  122.  
  123.  
  124.     # Add rule 
  125.     def add_rule_component(self, service, action, protocol, fromip, fromport, toip, toport):
  126.         # Component rule
  127.         if service == self.variable.get_constant("service_no"):
  128.             rule = "LANG=en ufw &action proto &protocol from &fromIP port &fromPort to &toIP port &toPort"
  129.         else:
  130.             rule = "LANG=en ufw &action &toPort"
  131.         
  132.         # Action
  133.         rule = rule.replace("&action", action)
  134.         
  135.         # Protocol
  136.         if protocol != self.variable.get_constant("both"):
  137.             rule = rule.replace("&protocol", protocol)
  138.         else:
  139.             rule = rule.replace(" proto &protocol ", " ")
  140.             
  141.         # FROM
  142.         if fromip != "":
  143.             rule = rule.replace("&fromIP", fromip)
  144.         else:
  145.             rule = rule.replace("&fromIP", self.variable.get_constant("any"))
  146.             
  147.         if fromport != "":
  148.             rule = rule.replace("&fromPort", fromport)
  149.         else:
  150.             rule = rule.replace(" port &fromPort ", " ")
  151.             
  152.         # TO
  153.         if toip != "":
  154.             rule = rule.replace("&toIP", toip)
  155.         else:
  156.             rule = rule.replace("&toIP", self.variable.get_constant("any"))
  157.             
  158.         if toport != "":
  159.             rule = rule.replace("&toPort", toport)
  160.         else:
  161.             rule = rule.replace(" port &toPort", "")
  162.  
  163.         # Return rule command
  164.         return rule
  165.  
  166.  
  167.     # Command Add rule
  168.     def add_rule(self, rule):
  169.         result = commands.getstatusoutput(rule)
  170.         return result[1]
  171.  
  172.  
  173.     # Add rule 
  174.     def remove_rule_component(self, rule_to, action, rule_from):
  175.         # Component rule
  176.         protocol = ""
  177.         toip     = ""
  178.         toport   = ""
  179.         fromip   = "" 
  180.         fromport = ""
  181.         rule   = "LANG=en ufw delete &action proto &protocol from &fromIP port &fromPort to &toIP port &toPort"
  182.         
  183.         # Set protocol
  184.         if rule_to.find("/tcp")   != -1 or rule_from.find("/tcp") != -1:
  185.             protocol = self.variable.get_constant("tcp")
  186.             rule_to   = rule_to.replace("/tcp","")
  187.             rule_from = rule_from.replace("/tcp","")
  188.         elif rule_to.find("/udp")   != -1 or rule_from.find("/udp") != -1:
  189.             protocol = self.variable.get_constant("udp")
  190.             rule_to   = rule_to.replace("/udp","")
  191.             rule_from = rule_from.replace("/udp","")
  192.         else:
  193.             protocol = self.variable.get_constant("both")
  194.  
  195.         
  196.         # Set IP/Port TO
  197.         to_aux = rule_to.split(" ")
  198.         if len(to_aux) == 1:
  199.             # IP or Port?
  200.             if to_aux[0] == self.variable.get_constant("anywhere"):
  201.                 toip = ""
  202.             elif ( to_aux[0].find(".") != -1 ) or ( len(to_aux[0]) > 11 ):
  203.                 toip = to_aux[0]
  204.             else:
  205.                 toport = to_aux[0]
  206.         elif len(to_aux) == 2:
  207.             toip   = to_aux[0]
  208.             toport = to_aux[1]
  209.         elif len(to_aux) == 3:
  210.             # It's a app command
  211.             rule = "LANG=en ufw delete " + to_aux[2]
  212.             return rule
  213.             
  214.         # Set IP/Port FROM
  215.         from_aux = rule_from.split(" ")
  216.         if len(from_aux) == 1:
  217.             # IP or Port?
  218.             if from_aux[0] == self.variable.get_constant("anywhere"):
  219.                 fromip = ""
  220.             elif ( from_aux[0].find(".") != -1 ) or ( len(from_aux[0]) > 11 ):
  221.                 fromip = from_aux[0]
  222.             else:
  223.                 fromport = from_aux[0]
  224.         elif len(from_aux) == 2:
  225.             fromip   = from_aux[0]
  226.             fromport = from_aux[1]
  227.         elif len(from_aux) == 3:
  228.             # It's a app command
  229.             rule = "LANG=en ufw delete " + from_aux[2]
  230.             return rule
  231.             
  232.         # Set rule    
  233.         # Action
  234.         rule = rule.replace("&action", action)
  235.         
  236.         # Protocol
  237.         if protocol != self.variable.get_constant("both"):
  238.             rule = rule.replace("&protocol", protocol)
  239.         else:
  240.             rule = rule.replace(" proto &protocol ", " ")
  241.             
  242.         # FROM
  243.         if fromip != "":
  244.             rule = rule.replace("&fromIP", fromip)
  245.         else:
  246.             rule = rule.replace("&fromIP", self.variable.get_constant("any"))
  247.             
  248.         if fromport != "":
  249.             rule = rule.replace("&fromPort", fromport)
  250.         else:
  251.             rule = rule.replace(" port &fromPort ", " ")
  252.             
  253.         # TO
  254.         if toip != "":
  255.             rule = rule.replace("&toIP", toip)
  256.         else:
  257.             rule = rule.replace("&toIP", self.variable.get_constant("any"))
  258.             
  259.         if toport != "":
  260.             rule = rule.replace("&toPort", toport)
  261.         else:
  262.             rule = rule.replace(" port &toPort", "")
  263.  
  264.         # Return rule command
  265.         return rule
  266.  
  267.  
  268.     # Command Add rule
  269.     def remove_rule(self, rule):
  270.         result = commands.getstatusoutput(rule)
  271.         return result[1]
  272.  
  273.  
  274.     # Return size Gufw window
  275.     def get_old_size_window(self):
  276.         if not self.variable.dev:
  277.             command = self.variable.get_command("read_size_win")
  278.         else:
  279.             command = self.variable.get_command("read_size_win_dev")
  280.             
  281.         width_height = commands.getstatusoutput(command)
  282.         if width_height[0] == 0:
  283.             width_height_split = width_height[1].split(";")
  284.             width  = width_height_split[0].replace("width=","")
  285.             height = width_height_split[1].replace("height=","")
  286.         else:
  287.             width  = self.variable.get_constant("window_width")
  288.             height = self.variable.get_constant("window_height")
  289.         
  290.         return int(width),int(height)
  291.         
  292.  
  293.     # Save actual size window
  294.     def save_size_window(self, win_width, win_height):
  295.         if not self.variable.dev:
  296.             command = self.variable.get_command("save_size_win")
  297.         else:
  298.             command = self.variable.get_command("save_size_win_dev")
  299.         
  300.         command = command.replace("&1", str(win_width))
  301.         command = command.replace("&2", str(win_height))
  302.         
  303.         commands.getstatusoutput(command)
  304.